#include <queue>
#include <vector>
#include <iostream>
using namespace std;
/*
给你一个大小为 m x n 的整数矩阵 isWater ，它代表了一个由 陆地 和 水域 单元格组成的地图。

如果 isWater[i][j] == 0 ，格子 (i, j) 是一个 陆地 格子。
如果 isWater[i][j] == 1 ，格子 (i, j) 是一个 水域 格子。
你需要按照如下规则给每个单元格安排高度：

每个格子的高度都必须是非负的。
如果一个格子是是 水域 ，那么它的高度必须为 0 。
任意相邻的格子高度差 至多 为 1 。当两个格子在正东、南、西、北方向上相互紧挨着，就称它们为相邻的格子。（也就是说它们有一条公共边）
找到一种安排高度的方案，使得矩阵中的最高高度值 最大 。

请你返回一个大小为 m x n 的整数矩阵 height ，其中 height[i][j] 是格子 (i, j) 的高度。如果有多种解法，请返回 任意一个 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/map-of-highest-peak
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
*/

class Solution {
public:
    vector<vector<int>> highestPeak(vector<vector<int>>& isWater) {
        int dir[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
        queue<pair<int,int>> q;
        int m=isWater.size(),n=isWater[0].size(),height=1;
        vector<vector<int>> res(m,vector<int>(n));
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
                if(isWater[i][j]==1)
                    q.push({i,j});
        while(q.size()){
            int size=q.size();
            while(size--){
                pair<int,int> t=q.front();
                q.pop();
                for(auto d:dir){    
                    int x=t.first+d[0],y=t.second+d[1];
                    if(x>=0&&x<m&&y>=0&&y<n&&isWater[x][y]==0&&res[x][y]==0){
                        res[x][y]=height;
                        q.push({x,y});
                    }
                }
            }
            height++;
        }
        return res;
    }
};